package com.atguigu.hbase;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.Cell;
import org.apache.hadoop.hbase.CellUtil;
import org.apache.hadoop.hbase.HBaseConfiguration;
import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.MasterNotRunningException;
import org.apache.hadoop.hbase.TableName;
import org.apache.hadoop.hbase.ZooKeeperConnectionException;
import org.apache.hadoop.hbase.client.Admin;
import org.apache.hadoop.hbase.client.Connection;
import org.apache.hadoop.hbase.client.ConnectionFactory;
import org.apache.hadoop.hbase.client.Delete;
import org.apache.hadoop.hbase.client.Get;
import org.apache.hadoop.hbase.client.Put;
import org.apache.hadoop.hbase.client.Result;
import org.apache.hadoop.hbase.client.ResultScanner;
import org.apache.hadoop.hbase.client.Scan;
import org.apache.hadoop.hbase.client.Table;
import org.apache.hadoop.hbase.util.Bytes;

/**
 * 
 * @author zhihu
 * HBaseAdmin(Admin): 管理表（创建，删除）
 * 		HTableDescriptor: 表描述器，用于创建表
 * 		HColumnDescriptor: 列描述器（构建列族）
 * 
 * Table: 用于表中数据的操作
 * 		Put: 用于封装待存放的数据
 * 		Delete: 用于封装待删除的数据
 * 		Get: 用于得到某一个具体的数据
 * 
 * Scan: 用于扫描表的配置信息
 * ResultScanner: 通过配置的扫描器，得到一个扫描表的实例扫描器
 * Result: 每一个该类型的实例化对象，都对应了一个rowkey中的若干数据
 * Cell: 用于封装一个rowkey下面所有单元格中的数据(rowkey, cf, cn, value)
 */
public class HBaseDemo {
	public static Configuration conf;

	static {
		// 使用 HBaseConfiguration 的单例方法实例化
		conf = HBaseConfiguration.create();
		conf.set("hbase.zookeeper.quorum", "hadoop3,hadoop4,hadoop5");
		conf.set("hbase.zookeeper.property.clientPort", "2181");
	}

	public static boolean isTableExist(String tableName)
			throws MasterNotRunningException, ZooKeeperConnectionException, IOException {
		/**
		 * 在 HBase 中管理、访问表需要先创建 HBaseAdmin对象
		 */

		// 老API
		// HBaseAdmin admin = new HBaseAdmin(conf);

		// 新API
		Connection connection = ConnectionFactory.createConnection(conf);
		Admin admin = connection.getAdmin();

		return admin.tableExists(TableName.valueOf(tableName));
	}

	// HBase表的创建
	public static void createTable(String tableName, String... columnFamily)
			throws MasterNotRunningException, ZooKeeperConnectionException, IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Admin admin = connection.getAdmin();
		// 判断表是否存在
		if (isTableExist(tableName)) {
			System.out.println("表" + tableName + "已存在");
			// System.exit(0);
		} else {
			// 创建表属性对象,表名需要转字节
			HTableDescriptor descriptor = new HTableDescriptor(TableName.valueOf(tableName));
			// 创建多个列族
			for (String cf : columnFamily) {
				descriptor.addFamily(new HColumnDescriptor(cf));
			} // 根据对表的配置，创建表
			admin.createTable(descriptor);
			System.out.println("表" + tableName + "创建成功！");
		}
	}

	// HBase表的删除
	public static void dropTable(String tableName)
			throws MasterNotRunningException, ZooKeeperConnectionException, IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Admin admin = connection.getAdmin();
		if (isTableExist(tableName)) {
			if (!admin.isTableDisabled(TableName.valueOf(tableName))) {
				admin.disableTable(TableName.valueOf(tableName));
			}
			admin.deleteTable(TableName.valueOf(tableName));
			System.out.println("表" + tableName + "删除成功！");
		} else {
			System.out.println("表" + tableName + "不存在！");
		}
	}

	// 添加一行数据
	public static void addRowData(String tableName, String rowKey, String columnFamily, String column, String value)
			throws IOException {
		// 创建 HTable 对象

		// 老API
		// HTable hTable = new HTable(conf, tableName);

		// 新API
		Connection connection = ConnectionFactory.createConnection(conf);
		Table table = connection.getTable(TableName.valueOf(tableName));

		// 向表中插入数据
		Put put = new Put(Bytes.toBytes(rowKey));
		// 向 Put 对象中组装数据

		// 老API
		// put.add(Bytes.toBytes(columnFamily), Bytes.toBytes(column),
		// Bytes.toBytes(value));

		put.addColumn(Bytes.toBytes(columnFamily), Bytes.toBytes(column), Bytes.toBytes(value));
		table.put(put);
		table.close();
		System.out.println("插入数据成功");
	}

	// 删除整行的所有列族、所有行、所有版本
	public static void deleteRow(String tableName, String rowKey) throws IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Table table = connection.getTable(TableName.valueOf(tableName));
		Delete delete = new Delete(Bytes.toBytes(rowKey));
		table.delete(delete);
		table.close();
	}

	// 删除多行数据
	public static void deleteMultiRow(String tableName, String... rows) throws IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Table table = connection.getTable(TableName.valueOf(tableName));
		List<Delete> deleteList = new ArrayList<Delete>();
		for (String row : rows) {
			Delete delete = new Delete(Bytes.toBytes(row));
			deleteList.add(delete);
		}
		table.delete(deleteList);
		table.close();
	}

	// 得到所有数据
	public static void getAllRows(String tableName) throws IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Table table = connection.getTable(TableName.valueOf(tableName));
		// 得到用于扫描 region 的对象
		Scan scan = new Scan();
		// 使用 HTable 得到 resultcanner 实现类的对象
		ResultScanner resultScanner = table.getScanner(scan);
		for (Result result : resultScanner) {
			Cell[] cells = result.rawCells();
			for (Cell cell : cells) {
				// 得到 rowkey
				System.out.println("行键:" + Bytes.toString(CellUtil.cloneRow(cell))); // 得到列族
				System.out.println("列族" + Bytes.toString(CellUtil.cloneFamily(cell)));
				System.out.println("列:" + Bytes.toString(CellUtil.cloneQualifier(cell)));
				System.out.println("值:" + Bytes.toString(CellUtil.cloneValue(cell)));
			}
		}
		table.close();
	}

	// 得到某一行所有数据
	public static void getRow(String tableName, String rowKey) throws IOException {
		Connection connection = ConnectionFactory.createConnection(conf);
		Table table = connection.getTable(TableName.valueOf(tableName));
		Get get = new Get(Bytes.toBytes(rowKey));
		// get.setMaxVersions();显示所有版本
		// get.setTimeStamp();显示指定时间戳的版本
		Result result = table.get(get);
		for (Cell cell : result.rawCells()) {
			System.out.println("行键:" + Bytes.toString(result.getRow()));
			System.out.println("列族" + Bytes.toString(CellUtil.cloneFamily(cell)));
			System.out.println("列:" + Bytes.toString(CellUtil.cloneQualifier(cell)));
			System.out.println("值:" + Bytes.toString(CellUtil.cloneValue(cell)));
			System.out.println("时间戳:" + cell.getTimestamp());
		}
	}

}
